• examples/vision/focal_modulation_network.py
  • Updated 'Last modified' date and added Keras 3 conversion attribution.
  • Replaced TensorFlow-specific imports with Keras 3 equivalent imports (keras, keras.ops).
  • Implemented environment variable settings to suppress TensorFlow C++ logging and Python warnings.
  • Refactored global configuration parameters, including BATCH_SIZE and EPOCHS.
  • Replaced tf.data pipeline with a custom FocalDataset class inheriting from keras.utils.PyDataset for backend-agnostic data handling.
  • Simplified PatchEmbed layer by removing redundant attributes and updating call method to use ops.shape.
  • Updated MLP function to use string literal 'gelu' for activation and removed type hints.
  • Corrected GeLU activation link in markdown comments.
  • Refactored FocalModulationLayer to simplify initialization, use ops.split and ops.softmax, and streamline context aggregation.
  • Simplified FocalModulationBlock and BasicLayer initializations and call methods, removing unused parameters and updating tensor operations to ops.
  • Integrated image preprocessing (rescaling, resizing, cropping, flipping) directly into the FocalModulationNetwork's call method.
  • Removed pos_drop and flatten layers from FocalModulationNetwork.
  • Modified display_grid function to use ops.convert_to_numpy and ops.norm for backend-agnostic visualization.
  • Updated TrainMonitor callback to use test_batch_images and integrate upsampler as a class attribute, removing intermediate image generation.
  • Adjusted WarmUpCosine learning rate schedule to use ops.cast and ops.where for backend compatibility.
  • Updated model compilation and fitting to use the new model instance and removed TrainMonitor callback from the fit call to prevent intermediate image generation.
  • Modified plotting code to create two subplots for loss and accuracy.
• examples/vision/ipynb/focal_modulation_network.ipynb
  • Updated 'Last modified' date in notebook metadata.
  • Reflected all code changes from focal_modulation_network.py within the notebook cells.
  • Removed output cells containing intermediate training progress and image visualizations.
• examples/vision/md/focal_modulation_network.md
  • Updated 'Last modified' date in the markdown header.
  • Reflected all code changes from focal_modulation_network.py within the markdown code blocks.
  • Removed image paths for intermediate training visualizations.

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.